/*
 * Copyright (c) 2011-2018, Meituan Dianping. All Rights Reserved.
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.dianping.cat.message.internal;

import com.dianping.cat.message.Message;
import com.dianping.cat.message.TaggedTransaction;
import com.dianping.cat.message.spi.MessageManager;
import com.dianping.cat.message.spi.MessageTree;

public class DefaultTaggedTransaction extends DefaultTransaction implements TaggedTransaction {
	private String m_rootMessageId;

	private String m_parentMessageId;

	private String m_tag;

	public DefaultTaggedTransaction(String type, String name, String tag, MessageManager manager) {
		super(type, name, manager);

		m_tag = tag;

		setStandalone(false);

		MessageTree tree = manager.getThreadLocalMessageTree();

		if (tree != null) {
			m_rootMessageId = tree.getRootMessageId();
			m_parentMessageId = tree.getMessageId();
		}
	}

	@Override
	public void bind(String tag, String childMessageId, String title) {
		DefaultEvent event = new DefaultEvent("RemoteCall", "Tagged");

		if (title == null) {
			title = getType() + ":" + getName();
		}

		event.addData(childMessageId, title);
		event.setTimestamp(getTimestamp());
		event.setStatus(Message.SUCCESS);
		event.setCompleted(true);

		addChild(event);
	}

	@Override
	public String getParentMessageId() {
		return m_parentMessageId;
	}

	@Override
	public String getRootMessageId() {
		return m_rootMessageId;
	}

	@Override
	public String getTag() {
		return m_tag;
	}

	@Override
	public void start() {
		MessageTree tree = getManager().getThreadLocalMessageTree();

		if (tree != null && tree.getRootMessageId() == null) {
			tree.setParentMessageId(m_parentMessageId);
			tree.setRootMessageId(m_rootMessageId);
		}
	}
}
